# 1.导入jsonpath
import jsonpath


class GetValue:
    @staticmethod
    def getValue(data: dict, keyword: str):
        """
        通过关键字获取值,默认获取第一个关键字字段的值
        :param data: 返回值
        :param keyword: 关键字
        :return:
        """
        try:
            return jsonpath.jsonpath(data, f"$..{keyword}")[0]  # 获取返回值中关键字字段的所有值
        except:
            print(f"{keyword}关键字不存在")  # 提示信息
            return None

    @staticmethod
    def getValues(data: dict, keyword: str):
        """
        通过关键字获取值
        :param data:
        :param keyword:
        :return:
        """
        try:
            return jsonpath.jsonpath(data, f"$..{keyword}")  # 获取返回值中关键字字段的所有值
        except:
            print(f"{keyword}关键字不存在")  # 提示信息
            return None
